package com.zxy.leetcode._00000_00099._00080_00089;

import java.util.Arrays;

/**
 * https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/
 *
 * 删除排序数组中的重复项 II
 *
 * 给定一个增序排列数组 nums ，你需要在 原地 删除重复出现的元素，
 * 使得每个元素最多出现两次，返回移除后数组的新长度。
 *
 * 不要使用额外的数组空间，你必须在 原地 修改输入数组
 * 并在使用 O(1) 额外空间的条件下完成。
 */
public class Test00080 {

    public static void main(String[] args) {
        Test00080 test = new Test00080();
        int[] nums = {0, 0, 1, 1, 1, 1, 2, 3, 3};
        int count = test.removeDuplicates(nums);
        System.out.println(Arrays.toString(nums));
        System.out.println(count);
    }

    public int removeDuplicates(int[] nums) {
        // w写方法的下标，r读方法的下标
        int w = 0;
        int r = 1;
        int len = nums.length;
        int num = nums[0];
        int count = 1;
        while (r < len) {
            if (nums[r] == num) {
                count++;
            }else {
                nums[w++] = num;
                // 控制最多写两个
                if (count > 1) {
                    nums[w++] = num;
                }
                num = nums[r];
                count = 1;
            }
            r ++;
        }

        nums[w++] = num;
        if (count > 1) {
            nums[w++] = num;
        }

        return w;
    }

    /*
    官方的解法
    总体思路是边读边写，最多写两个
    我的上面那套逻辑的思路是先读完再写，导致最后要再来一遍
     */
    public int removeDuplicates2(int[] nums) {
        int j = 1, count = 1;

        for (int i = 1; i < nums.length; i++) {
            if (nums[i] == nums[i - 1]) {
                count++;
            } else {
                count = 1;
            }
            if (count <= 2) {
                nums[j++] = nums[i];
            }
        }
        return j;
    }
}
